package com.xmtlxx.system.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.lang.tree.Tree;
import com.xmtlxx.common.core.domain.PageQuery;
import com.xmtlxx.common.core.domain.entity.SysArea;
import com.xmtlxx.common.core.page.TableDataInfo;
import com.xmtlxx.common.utils.StringUtils;
import com.xmtlxx.common.utils.TreeBuildUtils;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.xmtlxx.system.domain.bo.SysAreaBo;
import com.xmtlxx.system.domain.vo.SysAreaVo;
import com.xmtlxx.system.mapper.SysAreaMapper;
import com.xmtlxx.system.service.ISysAreaService;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

import java.util.Collection;
import java.util.List;
import java.util.Map;

/**
 * 行政区划管理Service业务层处理
 *
 * @author xmtlxx
 * @date 2022-04-28
 */
@Slf4j
@RequiredArgsConstructor
@Service
public class SysAreaServiceImpl implements ISysAreaService {

    private final SysAreaMapper baseMapper;

    /**
     * 查询行政区划管理
     *
     * @param areaId 行政区划管理主键
     * @return 行政区划管理
     */
    @Override
    public SysAreaVo queryById(String areaId){
        return baseMapper.selectVoById(areaId);
    }

    /**
     * 查询行政区划管理列表
     *
     * @param bo 行政区划管理
     * @return 行政区划管理
     */
    @Override
    public TableDataInfo<SysAreaVo> queryPageList(SysAreaBo bo, PageQuery pageQuery) {
        LambdaQueryWrapper<SysArea> lqw = buildQueryWrapper(bo);
        Page<SysAreaVo> result = baseMapper.selectVoPage(pageQuery.build(), lqw);
        return TableDataInfo.build(result);
    }

    /**
     * 查询行政区划管理列表
     *
     * @param bo 行政区划管理
     * @return 行政区划管理
     */
    @Override
    public List<SysAreaVo> queryList(SysAreaBo bo) {
        LambdaQueryWrapper<SysArea> lqw = buildQueryWrapper(bo);
        return baseMapper.selectVoList(lqw);
    }

    private LambdaQueryWrapper<SysArea> buildQueryWrapper(SysAreaBo bo) {
        Map<String, Object> params = bo.getParams();
        LambdaQueryWrapper<SysArea> lqw = Wrappers.lambdaQuery();
        lqw.like(StringUtils.isNotBlank(bo.getAreaName()), SysArea::getAreaName, bo.getAreaName());
        lqw.eq(bo.getPid() != null, SysArea::getPid, bo.getPid());
        lqw.like(StringUtils.isNotBlank(bo.getSname()), SysArea::getSname, bo.getSname());
        lqw.eq(bo.getParentId() != null, SysArea::getParentId, bo.getParentId());
        lqw.eq(bo.getLevels() != null, SysArea::getLevels, bo.getLevels());
        lqw.orderByAsc(SysArea::getAreaId);
        return lqw;
    }

    /**
     * 新增行政区划管理
     *
     * @param bo 行政区划管理
     * @return 结果
     */
    @Override
    public Boolean insertByBo(SysAreaBo bo) {
        SysArea add = BeanUtil.toBean(bo, SysArea.class);
        validEntityBeforeSave(add);

        //填充UUID
//        add.setAreaId(IdUtil.simpleUUID());
        if(bo.getPid() != null){
            SysArea area = baseMapper.selectById(bo.getPid());
            add.setLevels(area.getLevels()+1);
            add.setParentId(bo.getPid());
            add.setAncestors(area.getAncestors()+","+bo.getPid());
        }else{
            add.setLevels(1l);
            add.setParentId(0l);
            add.setPid(0l);
            add.setAncestors("0");
        }

        boolean flag = baseMapper.insert(add) > 0;
        if (flag) {
            bo.setAreaId(add.getAreaId());
        }
        return flag;
    }

    /**
     * 修改行政区划管理
     *
     * @param bo 行政区划管理
     * @return 结果
     */
    @Override
    public Boolean updateByBo(SysAreaBo bo) {
        SysArea update = BeanUtil.toBean(bo, SysArea.class);
        validEntityBeforeSave(update);
        return baseMapper.updateById(update) > 0;
    }

    /**
     * 保存前的数据校验
     *
     * @param entity 实体类数据
     */
    private void validEntityBeforeSave(SysArea entity){
        //TODO 做一些数据校验,如唯一约束
    }

    /**
     * 批量删除行政区划管理
     *
     * @param ids 需要删除的行政区划管理主键
     * @return 结果
     */
    @Override
    public Boolean deleteWithValidByIds(Collection<String> ids, Boolean isValid) {
        if(isValid){
            //TODO 做一些业务上的校验,判断是否需要校验
        }
        return baseMapper.deleteBatchIds(ids) > 0;
    }

    @Override
    public List<Tree<Long>> buildTreeSelect(List<SysAreaVo> list) {
        if (CollUtil.isEmpty(list)) {
            return CollUtil.newArrayList();
        }
        return TreeBuildUtils.build(list,(area, tree)->
            tree.setId(area.getAreaId())
                .setParentId(area.getParentId())
                .setName(area.getAreaName())
                .setWeight(area.getAreaId()));
    }

//    @Override
//    public void initDatas() {
//        Integer start = 11;
//        while (start <= 65){
//            SysArea area = baseMapper.selectById(start.toString());
//            if(area != null){
//
//
//            area.setLevels(1l);
//            area.setParentId(0l);
//            area.setAncestors("0");
//            area.setPid(0l);
//            baseMapper.updateById(area);
//
//            //二级
//            SysAreaBo secVo = new SysAreaBo();
//            secVo.setPid(area.getAreaId());
//            List<SysArea> secs = baseMapper.selectSysAreaList(secVo);
//            for(SysArea sec : secs){
//                sec.setLevels(2l);
//                sec.setParentId(sec.getPid());
//                sec.setAncestors(area.getAncestors()+","+sec.getPid());
//                baseMapper.updateById(sec);
//
//                //三级
//                SysAreaBo thdVo = new SysAreaBo();
//                thdVo.setPid(sec.getAreaId());
//                List<SysArea> thds = baseMapper.selectSysAreaList(thdVo);
//                for(SysArea thd : thds){
//                    thd.setLevels(3l);
//                    thd.setParentId(thd.getPid());
//                    thd.setAncestors(sec.getAncestors()+","+thd.getPid());
//                    baseMapper.updateById(thd);
//
//                    //四级
//                    SysAreaBo fthVo = new SysAreaBo();
//                    fthVo.setPid(thd.getAreaId());
//                    List<SysArea> fths = baseMapper.selectSysAreaList(fthVo);
//                    for(SysArea fth : fths){
//                        fth.setLevels(4l);
//                        fth.setParentId(fth.getPid());
//                        fth.setAncestors(thd.getAncestors()+","+fth.getPid());
//                        baseMapper.updateById(fth);
//                    }
//
//                }
//            }
//            }
//            start++;
//        }
//    }
}
